python itertools groupby multiple keys